MCPEditUserClient.py•4.47 kB
import socket
import ssl
import json
import argparse
import sys
def send_edit_user_request(server_ip, server_port, token, name=None, email=None, password=None, language=None,
                           timezone=None, roles=None, groups=None, use_public=False, activate_ftp=False, ftp_password=None,
                           use_ssl=True, accept_self_signed=False):
    """
    Sends a request to edit an existing user to the MCP server.
    """
    payload = {
        "command": "edit_user",
        "token": token,
        "arguments": {
            "name": name,
            "email": email,
            "password": password,
            "language": language,
            "timezone": timezone,
            "roles": roles or [],
            "groups": groups or [],
            "usePublic": use_public,
            "activateFtp": activate_ftp,
            "ftpPassword": ftp_password
        }
    }
    # Remove keys with None values
    payload["arguments"] = {k: v for k, v in payload["arguments"].items() if v is not None}
    payload_json = json.dumps(payload)
    
    raw_socket = None
    client_socket = None
    try:
        raw_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        raw_socket.settimeout(10)
        if use_ssl:
            context = ssl.create_default_context()
            if accept_self_signed:
                context.check_hostname = False
                context.verify_mode = ssl.CERT_NONE
            client_socket = context.wrap_socket(raw_socket, server_hostname=server_ip)
        else:
            client_socket = raw_socket
        
        client_socket.connect((server_ip, server_port))
        client_socket.sendall(payload_json.encode('utf-8'))
        response = b""
        while True:
            part = client_socket.recv(4096)
            if not part:
                break
            response += part
        return response.decode('utf-8')
    except ssl.SSLError:
        return "Error: Server and/or client may require TLS encryption. Please enable SSL/TLS."
    except Exception as e:
        return f"Error: {e}"
    
    finally:
        if client_socket is not None:
            try:
                client_socket.shutdown(socket.SHUT_RDWR)
            except:
                pass
            client_socket.close()
if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description="Send a request to edit an existing user to the MCP server.",
        formatter_class=argparse.RawTextHelpFormatter
    )
    parser.add_argument("--server-ip", required=True, help="IP address of the MCP server")
    parser.add_argument("--server-port", required=True, type=int, help="Port number of the MCP server")
    parser.add_argument("--token", required=True, help="Authentication token")
    parser.add_argument("--name", help="New name of the user")
    parser.add_argument("--email", help="New email of the user")
    parser.add_argument("--password", help="New password of the user")
    parser.add_argument("--language", help="Preferred language of the user")
    parser.add_argument("--timezone", help="Timezone of the user")
    parser.add_argument("--roles", nargs="*", help="List of roles to assign to the user")
    parser.add_argument("--groups", nargs="*", help="List of groups to assign to the user")
    parser.add_argument("--usePublic", action="store_true", help="Enable public knowledge base access")
    parser.add_argument("--activateFtp", action="store_true", help="Activate FTP access")
    parser.add_argument("--ftpPassword", help="Password for FTP access")
    parser.add_argument("--use-ssl", action="store_true", help="Connect using SSL/TLS")
    parser.add_argument("--accept-self-signed", action="store_true", help="Accept self-signed certificates (disable certificate verification)")
    if len(sys.argv) == 1:
        parser.print_help(sys.stderr)
        sys.exit(1)
    args = parser.parse_args()
    response = send_edit_user_request(
        args.server_ip,
        args.server_port,
        args.token,
        name=args.name,
        email=args.email,
        password=args.password,
        language=args.language,
        timezone=args.timezone,
        roles=args.roles,
        groups=args.groups,
        use_public=args.usePublic,
        activate_ftp=args.activateFtp,
        ftp_password=args.ftpPassword,
        use_ssl=args.use_ssl,
        accept_self_signed=args.accept_self_signed
    )
    print("Response from server:", response)